package com.itheima.utils;

import java.util.Random;

/**
 * @ClassName: ValidateCodeUtils
 * @Description: 随机生成短信验证码
 * @Author: 热爱生活の李
 * @Date: 2022/2/16 21:27
 */
public class ValidateCodeUtils {

    /*
    * @MethodName: generateValidateCode
    * @Description:随机生成4位或6位验证码
    * @Author: 热爱生活の李
    */
    public static Integer generateValidateCode(int length){
        Integer code =null;
        if(length == 4){
            code = new Random().nextInt(9999);//生成随机数，最大为9999
            if(code < 1000){
                code = code + 1000;//保证随机数为4位数字
            }
        }else if(length == 6){
            code = new Random().nextInt(999999);//生成随机数，最大为999999
            if(code < 100000){
                code = code + 100000;//保证随机数为6位数字
            }
        }else{
            throw new RuntimeException("只能生成4位或6位数字验证码");
        }
        return code;
    }


    /*
    * @MethodName: generateValidateCode4String
    * @Description: 随机生成指定长度的验证码
    * @Author: 热爱生活の李
    */
    public static String generateValidateCode4String(int length){
        Random rdm = new Random();
        String hash1 = Integer.toHexString(rdm.nextInt());
        String capstr = hash1.substring(0, length);
        return capstr;
    }
}
